import numpy as np
#将每一列的元素相加,将矩阵压缩为一行
print(np.sum([[0,1],[3, 5]],axis=0))
#将每一行的元素相加,将矩阵压缩为一列
print(np.sum([[0,1],[3, 5]],axis=1))